Class DenseEncoder

Inheritance Relationships

Base Type

Class Documentation

encoder_decoder_model.dense_encoder.DenseEncoder : public encoder_decoder_model.cnn_basenet.CNNBaseModel

Public Functions

__init__(self, l, n, growthrate, phase, with_bc=False, bc_theta=0.5)

Encoder based on DenseNet.

Parameters
  • l – According to the DenseNet paper L refers to the depth of the network

  • growthrate – According to the DenseNet paper GrowthRate refers to the dense block out dims

  • n – According to the DenseNet N refers to the block nums of the network

  • with_bc – whether to use BC in the model

  • phase – is training or testing

  • bc_theta – transmition theta thresholding

__str__(self)
Returns

encode(self, input_tensor, name)

DenseNet encoding.

Parameters
  • input_tensor

  • name

Returns